# -*- coding: UTF-8 -*-

from DetectorDeDiminutivos import *
from Tokenizador import *
import unittest

class detectorDeDiminutivosTest(unittest.TestCase):
	
	def setUp(self):
		self.detector = DetectorDeDiminutivos()
		self.tokenizador = Tokenizador()

	def test_publicable(self):
		text = 'Te mando un besito grande.'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Diminutivo', []))

	def test_detectarDiminutivo(self):
		text = '¡Tremendo boludín viniste a salir!'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Diminutivo', [('boludin', 'boludo')]))
		text = 'Es un boludito igual que vos'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Diminutivo', [('boludito', 'boludo')]))
		text = 'Que tipo mas pelotudin'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Diminutivo', [('pelotudin', 'pelotudo')]))
		text = 'soplaflautita tenias que ser marianita'
		tokens = self.tokenizador.tokenizar(text)
		self.assertEquals(self.detector.detectar(tokens), ('Diminutivo', [('soplaflautita', 'soplaflauta')]))	

if __name__ == "__main__": 
	unittest.main()
